"""
    Service agreement controller.
"""

from mnemoply import model
from mnemoply.controllers.lookup import LookupController, TableCol, TableLookupDataSource
from mnemoply.controllers.crud import CRUDController

class ServiceAgreementController(LookupController, CRUDController):
    model_object = model.ServiceAgreement
    lookup_columns = [
        TableCol(name='organization', caption=_('Organization')),
        TableCol(name='conclusion_date', caption=_('Conclusion date')),
        TableCol(name='customer', caption=_('Customer')),
        TableCol(name='service', caption=_('Service')),
        TableCol(name='department', caption=_('Department')),
        TableCol(name='cost', caption=_('Cost')),
    ]
    view_fields = ['organizationID',
                   'conclusionDate',
                   'customerID',
                   'serviceID',
                   'departmentID',
                   'cost',
                   ]

    def __init__(self):
        LookupController.__init__(self,
            data_source=TableLookupDataSource('service_agreement_lookup'),
            columns=self.lookup_columns,
            title=self.model_object.model_title,
        )
        CRUDController.__init__(self, model_object=self.model_object)

exposed = ServiceAgreementController()
